package top.milkbox.core;

import com.auth0.jwt.JWT;
import com.auth0.jwt.JWTCreator;
import com.auth0.jwt.algorithms.Algorithm;
import com.auth0.jwt.interfaces.Claim;
import com.auth0.jwt.interfaces.DecodedJWT;
import org.junit.Test;

import java.util.Date;
import java.util.HashMap;
import java.util.Map;

/**
 * JWT工具类
 * 此类依赖jar包：
 * <p>
 * <!--maven引入JWT-->
 * <dependency>
 * <groupId>com.auth0</groupId>
 * <artifactId>java-jwt</artifactId>
 * <version>3.10.0</version>
 * </dependency>
 *
 * @author https://blog.csdn.net/Top_L398/article/details/109361680 改编milkbox
 */
public class UtilsJWT {

    // 密匙，非常关键，不可泄密
    private static String SIGNATURE = "token!@#$%^7890";

    /**
     * 生成token
     *
     * @param map 传入payload，map中的数据会被加密存储在token中
     *            map类型必须为String, String
     * @return 返回token
     */
    public static String generate(Map<String, String> map) {
        JWTCreator.Builder builder = JWT.create();
//        map.forEach((k, v) -> {
//            builder.withClaim(k, v); // 这个函数有多个重构，可以传递很多类型
//        });
        map.forEach(builder::withClaim);
//        Calendar instance = Calendar.getInstance();
//        instance.add(Calendar.SECOND, 7);
//        builder.withExpiresAt(instance.getTime());
        builder.withExpiresAt(UtilsDate.dateAdd(new Date(), 1));
        return builder.sign(Algorithm.HMAC256(SIGNATURE));
    }

    /**
     * 验证token，通过抛异常的方式来判断token的状态
     * <p>
     * SignatureVerificationException 无效签名
     * <p>
     * TokenExpiredException token过期
     * <p>
     * AlgorithmMismatchException token算法不一致
     * <p>
     * Exception token失效或其他异常
     *
     * @param token token字符串
     */
    public static void verify(String token) {
        JWT.require(Algorithm.HMAC256(SIGNATURE)).build().verify(token);
    }

    /**
     * 获取token中payload
     * <p>
     * 解密出token中的所有内容，当token有问题的时候同样会抛出异常
     * <p>
     * SignatureVerificationException 无效签名
     * <p>
     * TokenExpiredException token过期
     * <p>
     * AlgorithmMismatchException token算法不一致
     * <p>
     * Exception token失效或其他异常
     *
     * @param token token
     * @return 返回值类型为DecodedJWT的实现类com.auth0.jwt.JWTDecoder
     */
    public static DecodedJWT analysis(String token) {
        return JWT.require(Algorithm.HMAC256(SIGNATURE)).build().verify(token);
    }


    /**
     * 生成token
     */
    @Test
    public void test1() {
        HashMap<String, String> map = new HashMap<>();
        map.put("username", "公孙轩辕");
        map.put("age", "30");

        String token = UtilsJWT.generate(map);
        System.out.println(token);
    }

    /**
     * 解析token
     */
    @Test
    public void test2() {
        DecodedJWT jwt = UtilsJWT.analysis("eyJ0eXAiOiJKV1QiLCJhbGciOiJIUzI1NiJ9.eyJleHAiOjE2ODU2MDcyMzgsImFnZSI6IjMwIiwidXNlcm5hbWUiOiLlhazlrZnovanovpUifQ._bDT4oD_WeBbGMa_9jy7jvn2IaatMz-BzpkvHiQoamM");
        Map<String, Claim> claims = jwt.getClaims();
        System.out.println("  键列表：" + claims.keySet().toString());
        System.out.println("过期时间：" + UtilsDate.dateToString(jwt.getClaim("exp").asDate(), "yyyy年MM月dd日 HH时mm分ss秒"));
        System.out.println("  用户名：" + jwt.getClaim("username").asString());
        System.out.println("    年龄：" + jwt.getClaim("age").asString());

    }
}
